package org.chnxi.algorithm.kmp;

import java.util.Arrays;

/**
 * KMP查找算法
 */
public class KMPMatch {

    public static void main(String[] args) {
        String str1 = "bbc abcdab abcdabcdabde";
        String str2 = "abcdabd";

        int[] next = kmpNext(str2);
        System.out.println("子串的匹配值表："+Arrays.toString(next));

        int idx = kmpMatch(str1 , str2);
        System.out.println("str2出现的位置，idx="+idx);

        String str = "pwwkew";
        System.out.println(Arrays.toString(kmpNext(str)));

    }

    /**
     * 使用kmp算法查找
     * @param str1 待匹配的字符串
     * @param str2 查找的字符串
     * @return 匹配到返回首次出现的索引位置，否则返回-1
     */
    public static int kmpMatch(String str1 , String str2){

        int[] next = kmpNext(str2);

        for (int i = 0, j = 0; i < str1.length(); i++){

            while(j > 0 && str1.charAt(i) != str2.charAt(j)){
                j = next[j-1];
            }

            if(str1.charAt(i) == str2.charAt(j)){
                j++;
            }
            if(j == str2.length()){
                return i - j + 1;
            }
        }

        return -1;

    }

    /**
     * 获取一个字符串（子串）的部分匹配值表
     */
    public static int[] kmpNext(String dest){
        //创建一个next数组保存部分匹配值
        int[] next = new int[dest.length()];
        //如果字符串长度为1，部分匹配值就是0
        next[0] = 0;

        for(int i=1 , j=0; i<dest.length(); i++){
            // 需要从next[j-1]获取新的j
            // 直到dest.charAt(i) == dest.charAt(j)后退出
            while(j>0 && dest.charAt(i) != dest.charAt(j)){
                j = next[j-1];
            }
            // 部分匹配值+1
            if(dest.charAt(i) == dest.charAt(j)){
                j++;
            }

            next[i] = j;
        }
        return next;
    }

}
